#!/bin/sh

[ -e /lib/firmware/$FIRMWARE ] && exit 0

. /lib/ar71xx.sh
. /lib/functions.sh
. /lib/functions/system.sh

ath9k_eeprom_die() {
	echo "ath9k eeprom: " "$*"
	exit 1
}

ath9k_eeprom_extract() {
	local part=$1
	local offset=$2
	local count=$3
	local mtd

	mtd=$(find_mtd_chardev $part)
	[ -n "$mtd" ] || \
		ath9k_eeprom_die "no mtd device found for partition $part"

	dd if=$mtd of=/lib/firmware/$FIRMWARE bs=1 skip=$offset count=$count 2>/dev/null || \
		ath9k_eeprom_die "failed to extract from $mtd"
}

ath9k_ubi_eeprom_extract() {
	local part=$1
	local offset=$2
	local count=$3
	local ubidev=$(nand_find_ubi $CI_UBIPART)
	local ubi

	ubi=$(nand_find_volume $ubidev $part)
	[ -n "$ubi" ] || \
		ath9k_eeprom_die "no UBI volume found for $part"

	dd if=/dev/$ubi of=/lib/firmware/$FIRMWARE bs=1 skip=$offset count=$count 2>/dev/null || \
		ath9k_eeprom_die "failed to extract from $ubi"
}

ath9k_patch_firmware_mac() {
	local mac=$1

	[ -z "$mac" ] && return

	macaddr_2bin $mac | dd of=/lib/firmware/$FIRMWARE conv=notrunc bs=1 seek=2 count=6
}

board=$(ar71xx_board_name)

case "$FIRMWARE" in
"soc_wmac.eeprom")
	case $board in
	c-55|\
	c-60)
		ath9k_eeprom_extract "art" 4096 2048
		ath9k_patch_firmware_mac $(macaddr_add $(mtd_get_mac_binary art 0) +1)
		;;
	mr18)
		. /lib/upgrade/nand.sh

		if [ -n "$(nand_find_volume ubi0 caldata)" ]; then
			ath9k_ubi_eeprom_extract "caldata" 4096 2048
		else
			ath9k_eeprom_extract "odm-caldata" 4096 2048
		fi
		ath9k_patch_firmware_mac $(macaddr_add $(mtd_get_mac_binary_ubi board-config 102) +1)
		;;
	r6100 | \
	wndr3700v4 | \
	wndr4300)
		ath9k_eeprom_extract "caldata" 4096 2048
		ath9k_patch_firmware_mac $(mtd_get_mac_binary caldata 0)
		;;
	z1)
		. /lib/upgrade/nand.sh

		if [ -n "$(nand_find_volume ubi0 caldata)" ]; then
			ath9k_ubi_eeprom_extract "caldata" 4096 2048
		else
			ath9k_eeprom_extract "origcaldata" 4096 2048
		fi
		ath9k_patch_firmware_mac $(macaddr_add $(mtd_get_mac_binary_ubi board-config 102) +2)
		;;
	*)
		ath9k_eeprom_die "board $board is not supported yet"
		;;
	esac
	;;

"pci_wmac0.eeprom")
	case $board in
	c-55)
		ath9k_eeprom_extract "art" 20480 2048
		ath9k_patch_firmware_mac $(macaddr_add $(mtd_get_mac_binary art 0) +2)
		;;
	mr18)
		. /lib/upgrade/nand.sh

		if [ -n "$(nand_find_volume ubi0 caldata)" ]; then
			ath9k_ubi_eeprom_extract "caldata" 20480 2048
		else
			ath9k_eeprom_extract "odm-caldata" 20480 2048
		fi
		ath9k_patch_firmware_mac $(macaddr_add $(mtd_get_mac_binary_ubi board-config 102) +2)
		;;
	wndr3700v4 | \
	wndr4300)
		ath9k_eeprom_extract "caldata" 20480 2048
		ath9k_patch_firmware_mac $(mtd_get_mac_binary caldata 12)
		;;
	z1)
		. /lib/upgrade/nand.sh

		if [ -n "$(nand_find_volume ubi0 caldata)" ]; then
			ath9k_ubi_eeprom_extract "caldata" 86016 4096
		else
			ath9k_eeprom_extract "origcaldata" 86016 4096
		fi
		ath9k_patch_firmware_mac $(macaddr_add $(mtd_get_mac_binary_ubi board-config 102) +3)
		;;
	*)
		ath9k_eeprom_die "board $board is not supported yet"
		;;
	esac
	;;

"pci_wmac1.eeprom")
	case $board in
	mr18)
		. /lib/upgrade/nand.sh

		if [ -n "$(nand_find_volume ubi0 caldata)" ]; then
			ath9k_ubi_eeprom_extract "caldata" 36864 2048
		else
			ath9k_eeprom_extract "odm-caldata" 36864 2048
		fi
		ath9k_patch_firmware_mac $(macaddr_add $(mtd_get_mac_binary_ubi board-config 102) +3)
		;;
	*)
		ath9k_eeprom_die "board $board is not supported yet"
		;;
	esac
	;;
esac
